package com.authine.cloudpivot.web.Utils;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.List;

/**
 * @author chenjq
 */
public class ImportUtil {

    /**
     * 2003- 版本的excel
     */
    private final static String excel2003L =".xls";

    /**
     * 2007+ 版本的excel
     */
    private final static String excel2007U =".xlsx";

    /**
     * 描述：获取IO流中的数据，组装成List<List<Object>>对象
     * @param in,fileName
     * param sheetNumber 获取第几个sheet的值，从0开始
     * @return
     * @throws Exception
     */
    public  List<List<Object>> getBankListByExcel(InputStream in, String fileName,Integer sheetNumber) throws Exception{
        List<List<Object>> list = null;

        //创建Excel工作薄
        Workbook work = this.getWorkbook(in,fileName);
        if(null == work){
            throw new Exception("创建Excel工作薄为空！");
        }
        Sheet sheet = null;
        Row row = null;
        Cell cell = null;

        list = new ArrayList<>();
        //遍历Excel中所有的sheet
        //for (int i = 0; i < work.getNumberOfSheets(); i++) {
            sheet = work.getSheetAt(sheetNumber);
            if(sheet==null){return null;}

            //遍历当前sheet中的所有行
            short cellNumber = 0;
            for (int j = sheet.getFirstRowNum(); j <= sheet.getLastRowNum(); j++) {
                row = sheet.getRow(j);
                //if(row==null||row.getFirstCellNum()==j){continue;}
                if(row == null) {continue;}

                //遍历所有的列
                //循环第一行，获取标题的列数
                if(j == 0) {
                    cellNumber = row.getLastCellNum();
                }

                List<Object> li = new ArrayList<Object>();
                for (int y = 0; y < cellNumber; y++) {
                    cell = row.getCell(y);
                    if(cell == null){
                        li.add(null);
                    }else {
                        li.add(this.getCellValue(cell));
                    }
                }
                list.add(li);
            }
        //}
        work.close();
        return list;
    }

    /**
     * 描述：根据文件后缀，自适应上传文件的版本
     * @param inStr,fileName
     * @return
     * @throws Exception
     */
    public  Workbook getWorkbook(InputStream inStr,String fileName) throws Exception{
        Workbook wb = null;
        String fileType = fileName.substring(fileName.lastIndexOf("."));
        if(excel2003L.equals(fileType)){
            //2003-
            wb = new HSSFWorkbook(inStr);
        }else if(excel2007U.equals(fileType)){
            //2007+
            wb = new XSSFWorkbook(inStr);
        }else{
            throw new Exception("解析的文件格式有误！");
        }
        return wb;
    }

    /**
     * 描述：对表格中数值进行格式化
     * @param cell
     * @return
     */
    public  Object getCellValue(Cell cell){
        Object value = null;
        //格式化number String字符
        DecimalFormat df = new DecimalFormat("0");
        //日期格式化
        SimpleDateFormat sdf = new SimpleDateFormat("yyy-MM-dd");
        //格式化数字
        DecimalFormat df2 = new DecimalFormat("0.00");
        DecimalFormat df1 = new DecimalFormat();
        df1.setGroupingUsed(false);

        CellType cellType = cell.getCellTypeEnum();

        switch (cellType){
            case STRING:
                if(cell.getRichStringCellValue().getString() != null){
                    value = String.format("%s",cell.getRichStringCellValue().getString());
                }
                break;
            case NUMERIC:
                if("General".equals(cell.getCellStyle().getDataFormatString())){
                    // df2.format(cell.getNumericCellValue());
                    value = df1.format(cell.getNumericCellValue());
                }else if("m/d/yy".equals(cell.getCellStyle().getDataFormatString())){
                    value = sdf.format(cell.getDateCellValue());
                }else{
                    // df2.format(cell.getNumericCellValue());
                    value = df1.format(cell.getNumericCellValue());
                }
                break;
            case BOOLEAN:
                value = cell.getBooleanCellValue();
                break;
            case BLANK:
                value = "";
                break;
            default:
                break;
        }
        return value;
    }

    /**
     * Excel导入
     *
     * @param file        文件
     * @param sheetNumber 获取第几个sheet的值，从0开始
     * @return
     */
    public static List<List<Object>> excelImport(MultipartFile file, Integer sheetNumber) throws IOException {
        ImportUtil importUtil = new ImportUtil();
        List<List<Object>> listob;
        if (file.isEmpty()) {
            throw new IOException("Excel不存在!");
        }
        try {
            InputStream in = file.getInputStream();
            listob = importUtil.getBankListByExcel(in, file.getOriginalFilename(), sheetNumber);
            in.close();
            return listob;
        } catch (Exception e) {
            throw new IOException("Excel导入失败!");
        }
    }
}
